package com.predict.javademo.mapper;

import com.predict.javademo.result.AdversarialResult;
import com.predict.javademo.result.DefenseResult;
import com.predict.javademo.result.OriginalResult;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Select;

import java.util.List;

@Mapper
public interface DataCollectMapper {

    @Select("SELECT label, class, pro FROM original")
    List<OriginalResult> collectOriginal();

    @Select("SELECT label, class, pro, img FROM adversarial")
    List<AdversarialResult> collectAdversarial();

    @Select("SELECT label, class, pro, img FROM defense")
    List<DefenseResult> collectDefense();

}
